cd ../..

gpus=$1

python -m trainer.BaselineTrainer fit \
    --config configs/experiment/baseline.yaml \
    --config configs/model/CRNN_nonlinear_cls.yaml \
    --config configs/dataset/desed.yaml \
    --config configs/trainer/default.yaml \
    --model.exp_name desed/new_baseline/non-linear-head \
    --trainer.devices=${gpus}